from setuptools import setup
from pybind11.setup_helpers import Pybind11Extension, build_ext

ext_modules = [
    Pybind11Extension(
        # "cpu_extension",
        "gpu_extension",
        ["mainCPU.cpp", "helloCPU.cpp"],
        extra_compile_args=["-O3"]
    ),
]

setup(
    # name="cpu_hello",
    name="gpu_extension",
    version="0.1",
    packages=[],  # ✅ 顯式指定沒有 Python 套件
    ext_modules=ext_modules,
    cmdclass={"build_ext": build_ext},
)